# encoding: utf-8
# 项目名称： drf-simple
# @File  : permission.py
# @Author: SunHao
# @Desc :  权限组件
# @Date  :   15:16 2025/03/16
from rest_framework.permissions import BasePermission
import random


class MyPermission(BasePermission):
    message = {"code": 600, "message": "您没有访问权限,请联系管理员"}

    def has_permission(self, request, view):
        # 获取请求中的数据，然后进行校验
        v1 = random.randint(1, 4)
        if v1 == 2:
            return True
        return False


class MyPermission1(BasePermission):
    message = {"code": 600, "message": "您没有访问权限,请联系管理员1"}

    def has_permission(self, request, view):
        # 获取请求中的数据，然后进行校验
        print("MyPermission1")
        return False


class MyPermission2(BasePermission):
    message = {"code": 600, "message": "您没有访问权限,请联系管理员2"}

    def has_permission(self, request, view):
        # 获取请求中的数据，然后进行校验
        print("MyPermission2")
        return False


class MyPermission3(BasePermission):
    message = {"code": 600, "message": "您没有访问权限,请联系管理员3"}

    def has_permission(self, request, view):
        # 获取请求中的数据，然后进行校验
        print("MyPermission3")
        return False


class UserPermission(BasePermission):
    message = {"code": 600, "message": "您没有访问权限,请联系管理员-user"}

    def has_permission(self, request, view):
        # 获取请求中的数据，然后进行校验
        print(request.user)
        if request.user.role == 3:
            return True
        return False


class ManagerPermission(BasePermission):
    message = {"code": 600, "message": "您没有访问权限,请联系管理员-Manager"}

    def has_permission(self, request, view):
        # 获取请求中的数据，然后进行校验
        if request.user.role == 2:
            return True
        return False


class BossPermission(BasePermission):
    message = {"code": 600, "message": "您没有访问权限,请联系管理员-Boss"}

    def has_permission(self, request, view):
        # 获取请求中的数据，然后进行校验
        if request.user.role == 1:
            return True
        return False
